package cn.bugstack.springframework.beans.factory.support;

import cn.bugstack.springframework.beans.BeansException;
import cn.bugstack.springframework.beans.PropertyValue;
import cn.bugstack.springframework.beans.PropertyValues;
import cn.bugstack.springframework.beans.factory.*;
import cn.bugstack.springframework.beans.factory.config.*;
import cn.bugstack.springframework.util.ClassUtils;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.util.StrUtil;

import java.lang.reflect.Constructor;
import java.lang.reflect.Method;

/**
 * 提供创建和管理Bean实例的抽象类，支持自动装配。
 * 继承自AbstractBeanFactory，增加了对Bean实例化策略的支持。
 */
public abstract class AbstractAutowireCapableBeanFactory extends AbstractBeanFactory implements AutowireCapableBeanFactory {

    // 实例化策略，默认使用JDK实例化策略
    // private InstantiationStrategy instantiationStrategy = new SimpleInstantiationStrategy();
    private InstantiationStrategy instantiationStrategy = new CglibSubclassingInstantiationStrategy();

    /**
     * 创建指定名称和定义的Bean实例。
     *
     * @param beanName Bean的名称。
     * @param beanDefinition Bean的定义。
     * @param args 创建Bean实例时使用的参数。
     * @return 创建的Bean实例。
     * @throws BeansException 如果实例化Bean失败。
     */
    @Override
    protected Object createBean(String beanName, BeanDefinition beanDefinition, Object[] args) throws BeansException {
        Object bean = null;
        try {
            /*
              与创建之前流程中的普通对象不同，代理对象的创建需要优于其他对象的创建。
              在实际开发中，优先判断Bean对象是否需要代理，如果需要，则直接返回代理对象。
             */
            bean = resolveBeforeInstantiation(beanName, beanDefinition);
            if (null != bean) {
                return bean;
            }
            // 实例化Bean
            bean = createBeanInstance(beanDefinition, beanName, args);
            // 实例化后判断
            boolean continueWithPropertyPopulation = applyBeanPostProcessorsAfterInstantiation(beanName, bean);
            if (!continueWithPropertyPopulation) {
                return bean;
            }
            // 在设置 Bean 属性之前，允许 BeanPostProcessor 修改属性值
            applyBeanPostProcessorsBeforeApplyingPropertyValues(beanName, bean, beanDefinition);
            // 填充Bean的属性
            applyPropertyValues(beanName, bean, beanDefinition);
            // 执行 Bean 的初始化方法和 BeanPostProcessor 的前置和后置处理方法
            bean = initializeBean(beanName, bean, beanDefinition);
        } catch (Exception e) {
            throw new BeansException("Instantiation of bean failed", e);
        }

        // 注册实现了 DisposableBean 接口或者在xml文件中配置了destory-method方法的 Bean 对象
        // 会将这些对方存放在DefaultSingletonBeanRegistry类中的Map<String, DisposableBean> disposableBeans中
        // 在虚拟机关闭时，会调用DefaultSingletonBeanRegistry中destroySingletons()方法，销毁这些注册的disposableBean
        registerDisposableBeanIfNecessary(beanName, bean, beanDefinition);

        /**
         * 注册为单例
         * AbstractBeanFactory继承了DefaultSingletonBeanRegistry，而该类又实现了SingletonBeanRegistry接口
         * 将创建出来的单例Bean，缓存到DefaultSingletonBeanRegistry类中的Map<String, Object> singletonObjects = new HashMap<>();
         * 判断 SCOPE_SINGLETON、SCOPE_PROTOTYPE
         * 单例模式（Singleton）和原型模式的区别为是否将Bean对象存储到内存中。
         * 如果是原型模式，就不会将Bean对象存储到内存中，每次获取都需要重新创建对象。
         * 非单例模式的Bean不需要执行销毁方法。
         */
        if (beanDefinition.isSingleton()) {
            addSingleton(beanName, bean);
        }

        return bean;
    }

    /**
     * Bean 实例化后对于返回 false 的对象，不再执行后续设置 Bean 对象属性的操作
     *
     * @param beanName
     * @param bean
     * @return
     */
    private boolean applyBeanPostProcessorsAfterInstantiation(String beanName, Object bean) {
        boolean continueWithPropertyPopulation = true;
        for (BeanPostProcessor beanPostProcessor : getBeanPostProcessors()) {
            if (beanPostProcessor instanceof InstantiationAwareBeanPostProcessor) {
                InstantiationAwareBeanPostProcessor instantiationAwareBeanPostProcessor = (InstantiationAwareBeanPostProcessor) beanPostProcessor;
                if (!instantiationAwareBeanPostProcessor.postProcessAfterInstantiation(bean, beanName)) {
                    continueWithPropertyPopulation = false;
                    break;
                }
            }
        }
        return continueWithPropertyPopulation;
    }


    /**
     * 在实例化Bean之前解析处理。
     * 这个方法会应用所有注册了的BeanPostProcessor的postProcessBeforeInstantiation方法。
     * 如果任何一个BeanPostProcessor返回了一个非null的对象，那个对象将被作为Bean的实例返回，不再继续实例化原Bean。
     *
     * @param beanName Bean的名称
     * @param beanDefinition Bean的定义
     * @return 如果有任何一个BeanPostProcessor在实例化前处理返回了非null值，则返回那个值；否则返回null。
     */
    protected Object resolveBeforeInstantiation(String beanName, BeanDefinition beanDefinition) {
        // 应用所有BeanPostProcessor的postProcessBeforeInstantiation方法
        Object bean = applyBeanPostProcessorsBeforeInstantiation(beanDefinition.getBeanClass(), beanName);
        // 如果实例化前的处理返回了非null对象，则对其进行初始化后的处理
        if (null != bean) {
            bean = applyBeanPostProcessorsAfterInitialization(bean, beanName);
        }
        return bean;
    }

    /**
     * 应用所有InstantiationAwareBeanPostProcessor的postProcessBeforeInstantiation方法。
     * 这个方法会遍历所有的BeanPostProcessor，如果其中任何一个实现了InstantiationAwareBeanPostProcessor接口，
     * 则调用其postProcessBeforeInstantiation方法，并返回该方法的返回值（如果非null）。
     *
     * @param beanClass 要实例化的Bean的类
     * @param beanName Bean的名称
     * @return 如果有任何一个InstantiationAwareBeanPostProcessor的postProcessBeforeInstantiation方法返回了非null值，则返回那个值；否则返回null。
     */
    protected Object applyBeanPostProcessorsBeforeInstantiation(Class<?> beanClass, String beanName) {
        for (BeanPostProcessor beanPostProcessor : getBeanPostProcessors()) {
            // 如果BeanPostProcessor实现了InstantiationAwareBeanPostProcessor接口，则调用其postProcessBeforeInstantiation方法
            if (beanPostProcessor instanceof InstantiationAwareBeanPostProcessor) {
                Object result = ((InstantiationAwareBeanPostProcessor) beanPostProcessor).postProcessBeforeInstantiation(beanClass, beanName);
                // 如果处理返回了非null对象，则直接返回该对象
                if (null != result) return result;
            }
        }
        return null;
    }

    /**
     * 在设置 Bean 属性之前，允许 BeanPostProcessor 接口修改属性值
     *
     * @param beanName
     * @param bean
     * @param beanDefinition
     */
    protected void applyBeanPostProcessorsBeforeApplyingPropertyValues(String beanName, Object bean, BeanDefinition beanDefinition) {
        for (BeanPostProcessor beanPostProcessor : getBeanPostProcessors()) {
            // 检查当前的 Bean 后处理器是否属于 InstantiationAwareBeanPostProcessor 接口的实现类。
            // 这个接口提供了在实例化 Bean 之前和之后的扩展点，通常用于自定义实例化逻辑或者修改属性值。
            if (beanPostProcessor instanceof InstantiationAwareBeanPostProcessor){
                PropertyValues pvs = ((InstantiationAwareBeanPostProcessor) beanPostProcessor).postProcessPropertyValues(beanDefinition.getPropertyValues(), bean, beanName);
                // 如果 postProcessPropertyValues 返回的 PropertyValues 对象不是 null，
                // 则遍历这个对象中的所有 PropertyValue，并将它们添加到 beanDefinition 的 PropertyValues 中。
                if (null != pvs) {
                    for (PropertyValue propertyValue : pvs.getPropertyValues()) {
                        beanDefinition.getPropertyValues().addPropertyValue(propertyValue);
                    }
                }
            }
        }
    }


    /**
     * 给Bean填充属性值。
     *
     * @param beanName Bean的名称。
     * @param bean Bean实例。
     * @param beanDefinition Bean的定义。
     */
    private void applyPropertyValues(String beanName, Object bean, BeanDefinition beanDefinition) {
        try {
            // 获取并应用Bean的属性值
            PropertyValues propertyValues = beanDefinition.getPropertyValues();
            for (PropertyValue propertyValue : propertyValues.getPropertyValues()) {

                String name = propertyValue.getName();
                Object value = propertyValue.getValue();

                // 如果属性值是Bean引用，则解析为对应的Bean实例
                if (value instanceof BeanReference) {
                    BeanReference beanReference = (BeanReference) value;
                    value = getBean(beanReference.getBeanName());
                }
                // 设置属性值
                BeanUtil.setFieldValue(bean, name, value);
            }
        } catch (Exception e) {
            throw new BeansException("Error setting property values：" + beanName);
        }
    }

    /**
     * 创建Bean实例。
     *
     * @param beanDefinition Bean的定义。
     * @param beanName Bean的名称。
     * @param args 创建Bean实例时使用的参数。
     * @return 创建的Bean实例。
     */
    protected Object createBeanInstance(BeanDefinition beanDefinition, String beanName, Object[] args) {
        Constructor constructorToUse = null;
        Class<?> beanClass = beanDefinition.getBeanClass();
        // 处理无参构造函数的
        if (null == args) {
//            try {
//                constructorToUse = beanClass.getDeclaredConstructor();
//            } catch (NoSuchMethodException e) {
//                throw new BeansException(beanName + "don't have noargs constructor, but you don't convey args");
//            }
            // 这种情况默认constructorToUse为Null，在InstantiationStrategy中
            // 会对该情况进行判断，以方便直接调用无参构造方法创建对象
        }
        else {  // 处理有参构造的
            Constructor<?>[] declaredConstructors = beanClass.getDeclaredConstructors();
            for (Constructor constructor : declaredConstructors) {
                Class<?>[] parameterTypes = constructor.getParameterTypes();
                if (parameterTypes.length == args.length) {
                    boolean parametersMatch = true;
                    // 主要用于检查构造函数的参数类型是否与传入的参数类型匹配。
                    for (int i = 0; i < parameterTypes.length; i++) {
                        // isAssignableFrom(Class<?> cls) 方法是 Java 中的一个方法，
                        // 它用于检查当前 Class 对象是否可以赋值给参数指定的 Class 对象。如果可以，则返回 true；否则返回 false。
                        if (!parameterTypes[i].isAssignableFrom(args[i].getClass())) {
                            parametersMatch = false;
                            break;
                        }
                    }
                    // 如果参数类型匹配，尝试使用该构造函数实例化对象
                    if (parametersMatch) {
                        constructorToUse = constructor;
                        break;
                    }
                }
            }
            if (null == constructorToUse) {
                throw new BeansException("your args have error, don't have matched constructor");
            }
        }
        // 使用选择的构造函数实例化Bean
        return instantiationStrategy.instantiate(beanDefinition, beanName, constructorToUse, args);
    }

    /**
     * 获取当前的实例化策略。
     *
     * @return 当前的实例化策略。
     */
    public InstantiationStrategy getInstantiationStrategy() {
        return instantiationStrategy;
    }

    /**
     * 设置实例化策略。
     *
     * @param instantiationStrategy 要设置的实例化策略。
     */
    public void setInstantiationStrategy(InstantiationStrategy instantiationStrategy) {
        this.instantiationStrategy = instantiationStrategy;
    }

    /**
     * 初始化Bean实例。
     * 首先，此方法会执行BeanPostProcessor接口的before初始化方法，
     * 然后调用Bean的初始化方法（如果定义了的话），
     * 最后执行BeanPostProcessor接口的after初始化方法。
     *
     * @param beanName Bean的名称。
     * @param bean Bean实例。
     * @param beanDefinition Bean的定义信息。
     * @return 经过BeanPostProcessor处理后的Bean实例。
     */
    private Object initializeBean(String beanName, Object bean, BeanDefinition beanDefinition) {

        // invokeAwareMethods
        // 不能直接在创建Bean时获取到ApplicationContext属性
        if (bean instanceof Aware) {
            if (bean instanceof BeanFactoryAware) {
                // 在Java中，关键字this表示当前对象的引用。在一个抽象类中，this同样指代当前对象，即使用这个抽象类派生出来的具体对象。
                ((BeanFactoryAware) bean).setBeanFactory(this);
            }
            if (bean instanceof BeanClassLoaderAware) {
                ((BeanClassLoaderAware) bean).setBeanClassLoader(getBeanClassLoader());
            }
            if (bean instanceof BeanNameAware) {
                ((BeanNameAware) bean).setBeanName(beanName);
            }
        }

        // 1. 执行 BeanPostProcessor Before 前置处理
        Object wrappedBean = applyBeanPostProcessorsBeforeInitialization(bean, beanName);

        // 执行 Bean 对象的初始化方法
        // 如果没有通过接口或者xml文件中的注解定义init-method，则该方法其实是一个空的方法
        try {
            invokeInitMethods(beanName, wrappedBean, beanDefinition);
        } catch (Exception e) {
            throw new BeansException("Invocation of init method of bean[" + beanName + "] failed", e);
        }

        // 2. 执行 BeanPostProcessor After 后置处理
        wrappedBean = applyBeanPostProcessorsAfterInitialization(wrappedBean, beanName);
        return wrappedBean;
    }


    /**
     * 在Bean初始化之前应用所有的BeanPostProcessor处理器。
     * 通过getBeanPostProcessor()获得到的 List<BeanPostProcessor> beanPostProcessors中，已经存放了已经注册的该类型的Bean，
     * 注册是在AbstractApplicationContext抽象类的refresh方法中注册的
     * // 5. BeanPostProcessor 需要提前于其他 Bean 对象实例化之前执行注册操作
     * registerBeanPostProcessors(beanFactory);
     *
     * @param existingBean 待处理的Bean实例。
     * @param beanName Bean的名称。
     * @return 经过所有BeanPostProcessor处理器处理后的Bean实例。
     * @throws BeansException 如果处理过程中发生错误。
     */
    @Override
    public Object applyBeanPostProcessorsBeforeInitialization(Object existingBean, String beanName) throws BeansException {
        Object result = existingBean; // 初始化结果为传入的Bean实例
        for (BeanPostProcessor processor : getBeanPostProcessors()) { // 遍历所有BeanPostProcessor处理器
            Object current = processor.postProcessBeforeInitialization(result, beanName); // 应用当前处理器
            if (null == current) {
                return result; // 如果当前处理器返回null，结束遍历并返回原始Bean实例
            }
            result = current; // 更新结果为当前处理器返回的实例
        }
        return result; // 返回经过所有处理器处理后的Bean实例
    }


    /**
     * 在Bean初始化之后应用BeanPostProcessor处理器。
     *
     * @param existingBean 已存在并已初始化的Bean实例。
     * @param beanName Bean的名称。
     * @return 经过所有BeanPostProcessor处理后的Bean实例。
     * @throws BeansException 如果处理过程中发生错误。
     */
    @Override
    public Object applyBeanPostProcessorsAfterInitialization(Object existingBean, String beanName) throws BeansException {
        Object result = existingBean; // 初始化结果为传入的Bean实例
        for (BeanPostProcessor processor : getBeanPostProcessors()) {
            // 遍历所有BeanPostProcessor并应用它们的postProcessAfterInitialization方法
            Object current = processor.postProcessAfterInitialization(result, beanName);
            if (null == current) {
                // 如果某个处理器返回null，则停止处理并返回当前已处理的Bean实例
                return result;
            }
            // 将当前处理器处理后的结果作为下一个处理器的输入
            result = current;
        }
        return result; // 返回经过所有处理器处理后的Bean实例
    }


    /**
     * 调用 Bean 的初始化方法。
     * 在 Spring 中，Bean 的初始化方法可以通过多种方式指定，例如在 JavaConfig 中使用 @Bean(initMethod = "init") 注解，
     * 或者在 XML 配置文件中使用 <bean> 标签的 init-method 属性指定，或者继承实现InitializingBean接口来实现初始化逻辑。
     * 本方法负责执行这些指定的初始化方法的逻辑。
     *
     * @param beanName Bean 的名称。
     * @param bean 封装了 Bean 实例的对象。
     * @param beanDefinition Bean 的定义信息，包含了初始化方法等配置信息。
     */
    private void invokeInitMethods(String beanName, Object bean, BeanDefinition beanDefinition) throws Exception {
        // 1. 实现接口 InitializingBean
        if (bean instanceof InitializingBean) {
            ((InitializingBean) bean).afterPropertiesSet();
        }

        // 2. 注解配置 init-method {判断是为了避免二次执行初始化}
        String initMethodName = beanDefinition.getInitMethodName();
        if (StrUtil.isNotEmpty(initMethodName) && !(bean instanceof InitializingBean)) {
            Method initMethod = beanDefinition.getBeanClass().getMethod(initMethodName);
            if (null == initMethod) {
                throw new BeansException("Could not find an init method named '" + initMethodName + "' on bean with name '" + beanName + "'");
            }
            initMethod.invoke(bean);
        }
    }

    /**
     * 在创建Bean对象的实例时，需要将销毁方法保存起来，方便销毁时调用。
     * DisposableBean销毁方法的具体信息会被注册到DefaultSingletonBeanRegistry中新增的Map disposableBeans属性中，
     * 因为DisposableBean接口的方法最终可以被AbstractApplicationContext类的close方法通过getBeanFactory().destorySingletons来调用。
     */
    protected void registerDisposableBeanIfNecessary(String beanName, Object bean, BeanDefinition beanDefinition) {
        // 非 Singleton 类型的 Bean 不执行销毁方法
        if (!beanDefinition.isSingleton()) return;

        if (bean instanceof DisposableBean || StrUtil.isNotEmpty(beanDefinition.getDestroyMethodName())) {
            registerDisposableBean(beanName, new DisposableBeanAdapter(bean, beanName, beanDefinition));
        }
    }
}

